性能优化技巧：TopN

TopN是常见的运算，用SQL写出来是这样（以Oracle为例）：

select * from (select * from T order by x desc) where rownum<=N 

这个SQL的运算逻辑从其语句上看，要先做排序（Order by），然后再取出前N条。

我们知道，排序是个非常慢的动作，复杂度很高（n*logn），如果涉及数据量大到内存放不下，那还需要进行内外存数据交换，性能还会急剧下降。

而事实上，要计算TopN，我们能设计出不需要全排序的算法，只要保持一个大小为N的小集合，对数据集遍历一次，将遍历过的数据的前N名保存在这个小集合中，遍历到新一条数据，如果比当前的第N名还大，则插入进去并丢掉现在的第N名，如果比当前的第N名要小，则不做动作。

这样计算的复杂度要低很多（n*logN，n是总数据条数），而且一般N都不大，可以在内存中放下，无论数据量有多大，都不会涉及内外存交换问题。

但是，SQL无法描述上面这个计算过程，这时，我们只能寄希望于数据库引擎是不是能自己优化。使用SPL就容易描述上面这个计算过程，从而实现高性能计算。

我们来测试一下看Oracle是不是会做这个优化，即用Oracle实现TopN后和SPL做同样运算相比。因为SPL能使用优化算法，如果Oracle的计算时间和SPL差不多，则说明它做了优化，如果差得很远，则可能是做了全排序。

 一、数据准备和环境 

用SPL脚本生成数据文件，数据共两列，第一列id是小于20亿的随机整数，第二列amount是不大于1千万的随机实数。数据记录为80亿行，生成的原始文本文件大小为169G。利用数据库提供的数据导入工具将此文件数据导入到Oracle的数据表topn中，同时也用此文件数据生成SPL组表文件topn.ctx。

在一台Intel服务器上完成测试，2个Intel3014 CPU，主频1.7G，共12核，内存64G。数据库表数据及SPL组表文件均存储在同一块SSD硬盘上。

我们刻意将数据量设计得比内存大，这样如果进行排序则会有内外存交换动作，性能下降会非常大，容易被观测到。

 二、常规TopN

取出topn表中amount最大的前100条。

1.  Oracle测试

查询用的SQL语句是：

select * from (
    select  /*+ parallel(4) */
    * from topn order by amount desc
) where rownum<=100;

说明：/*+ parallel(4) */ 是Oracle的并行查询语法，其中4是并行数。

2. SPL测试

编写SPL脚本执行测试：

	A
1	=now()	/记录时间
2	=4	/并行数
3	=file("/home/topn/topn.ctx").open()	/产生组表对象
4	=A3.cursor@m(id,amount;;A2).groups(;top(100;-amount))
5	=interval@s(A1,now())	/算出执行时间

与SQL不同，SPL把TopN看成是一种聚合运算，和sum/count这类运算一样，不同的只是TopN将返回一个集合，而sum/count返回的是单值。但它们的计算逻辑是一样的，都只需要对原数据集遍历一次，并且不涉及全排序。

A4格中的groups(;top(100;-amount)就是对全集做TopN聚合运算，计算出前100名。

3.  结论与分析

常规TopN测试时间见下表：

测试结果（时间单位：秒）

并行数	1	2	4	8	12
Oracle	1922	952	526	308	256
SPL组表	2641	1565	729	371	321

测试表明，Oracle比SPL还要更快一点，而SPL没有做全排序，这说明Oracle在这种情况会自动优化。

Oracle比SPL快也可以理解，因为Oracle是用C++开发的，而目前版本的SPL是用java开发的。C++比java计算更快是正常的，而且这次测试读取了全部的两列数据，且数据随机无序，很难压缩，所以组表的列式存储也没有优势。

三、增加复杂度

对于最基本的TopN，Oracle很聪明，即使写成了子查询也会优化。我们下面增加问题的难度，改成分组后在每一组中做TopN。

具体运算设计为：按照id字段最后一位数字值进行分组，然后查询出每组中amount最大的前100条记录。id是个整数，这样也就只有10个分组，分组本身的计算量并不大，不过要对分组数据做TopN，整体计算复杂度略高了一些。如果没有全排序，总体运算时间应当比刚才的情况多，但还在同一数量级范围内。

1. Oracle测试

查询用的SQL语句是：

select * from (
    select  /*+ parallel(2) */
    mod(id,10) as gid,amount,
    row_number()over (partition by mod(id,10) order by amount desc) rn
    from topn
) where rn <= 100;

SQL无法直接写出分组后取TopN的运算，只能借助窗口函数计算序号，语法中还是有排序（order by）的字样。

2. SPL组表测试

编写SPL脚本执行测试：

	A
1	=now()	/记录时间
2	=4	/并行数
3	=file("/home/topn/topn.ctx").open()	/产生组表对象
4	=A3.cursor@m(id,amount;;A2).groups@u(id%10:gid;top(100;-amount))
5	=interval@s(A1,now())	/算出执行时间

因为SPL把TopN看成是聚合计算，所以可以很容易放在分组汇总中，和全量聚合的写法几乎是一样的。

3. 结论与分析

测试结果（时间单位：秒）

并行数	1	2	4	8	12
Oracle	41649	19602	9359	4627	3211
SPL组表	4380	2127	1007	465	349

增加难度后，Oracle比之前的简单情况慢了10倍还多，而且比SPL做同样运算也慢了近10倍。这说明Oracle在这种情况下很可能做了排序动作，情况复杂化之后，Oracle的优化引擎不起作用了。

而SPL在这两种情况下的运算时间差别不到2倍，基本上在同一数量级，符合我们之前的分析，算法的优势得到了充分的体现。

更多性能优化技巧，可在底部“阅读原文”中查看

重磅！开源SPL交流群成立了

简单好用的SPL开源啦！

为了给感兴趣的小伙伴们提供一个相互交流的平台，

特地开通了交流群（群完全免费，不广告不卖课）

需要进群的朋友，可长按扫描下方二维码

本文感兴趣的朋友，请转到阅读原文去收藏 ^_^